# Copyright (C) 2011 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Main webapp driver for spojtweet."""

__author__ = 'ricbit@google.com (Ricardo Bittencourt)'

import datetime
import os
import urllib2

from google.appengine.dist import use_library
use_library('django', '1.2')

from google.appengine.ext import webapp
from google.appengine.ext.webapp import template
from google.appengine.ext.webapp.util import run_wsgi_app
from google.appengine.ext import db

import admin
import crawler
import model
import parser
import refresh_user_page
import twitter

class MainPage(webapp.RequestHandler):
  def get(self):
    self.response.out.write('Under construction')

class UserPage(webapp.RequestHandler):
  def get(self, user):
    spojuser = model.SpojUser.get_by_key_name(user)
    if (spojuser is None or
        spojuser.version is None or
	spojuser.version < model.VERSION):
      self.response.out.write(
          'Please <a href="/refresh/%s">refresh user %s</a>' % (user, user))
      return
    path = os.path.join(os.path.dirname(__file__), 'user.html')
    badge_value_names = {
      1: 'gold',
      2: 'silver',
      3: 'bronze'
    }
    badges = spojuser.badges[:]
    badges.sort(key=lambda x: (x.value, x.name))
    for badge in badges:
      badge.value = badge_value_names[badge.value]
    values = {
      'name': spojuser.name,
      'country': spojuser.country.title(),
      'badges': badges
    }
    self.response.out.write(template.render(path, values))

class NotFoundPage(webapp.RequestHandler):
  def get(self):
    self.error(404)
    self.response.out.write('404 Not Found')

app = webapp.WSGIApplication(
          [('/', MainPage),
	   ('/user/([^/]+)', UserPage),
	   ('/refresh/([^/]+)', refresh_user_page.RefreshUserPage),
	   ('/crawl', crawler.CrawlCountryPage),
	   ('/admin', admin.AdminPage),
	   ('/admin/keys/(.*?)/(.*?)/', admin.SetKeyPage),
	   ('/twitter', twitter.TwitterPage),
	   ('/twitter/auth/(.*?)/', twitter.TwitterAuthPage),
	   ('/twitter/send/(.*?)/', twitter.SendTweetPage),
	   ('/.*', NotFoundPage)],
	  debug=True)

def main():
  run_wsgi_app(app)

if __name__ == '__main__':
  main()
